import os

CWD = os.path.dirname(__file__)

BERT_HOST = '172.16.133.231'
if os.getenv('BERT_HOST'):
    BERT_HOST = os.getenv('BERT_HOST')
print('BERT_HOST:', BERT_HOST)

BERT_PORT = 5555
BERT_DIMENSION = 768

MILVUS_HOST = "1.116.234.26"
MILVUS_PORT = 19530
MILVUS_QA_TABLE = "qa_system_test"

KNOWLEDGE_FILE_NAME = 'data/data.csv'

if os.getenv('KNOWLEDGE_FILE_PATH'):
    KNOWLEDGE_FILE_PATH = os.getenv('KNOWLEDGE_FILE_PATH')
else:
    KNOWLEDGE_FILE_PATH = os.path.join(CWD, KNOWLEDGE_FILE_NAME)
print('KNOWLEDGE_FILE_PATH:', KNOWLEDGE_FILE_PATH)

MYSQL_IP = '172.16.20.157'
MYSQL_PORT = 4000
MYSQL_USER = 'admin'
MYSQL_PASSWORD = 'TSPxg123dgg'
MYSQL_QA_TABLE = "qa_system_test_algorithm"
MYSQL_QA_DB = "bigdata_platform"

ANSWER_TOP_K = 1
